Image classification project in python